package com.itheima.controller;


import com.itheima.dto.MessageConstant;
import com.itheima.dto.Result;
import com.itheima.service.MemberService;
import com.itheima.service.OrderService;
import com.itheima.service.ReportService;
import com.itheima.utils.DateUtils;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.ResourceUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.File;
import java.io.FileInputStream;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * 报表操作
 */
@RestController
@RequestMapping("/report")
public class ReportController {
    @Autowired
    private MemberService memberService;
    @Autowired
    private ReportService reportService;

    /**
     * 会员数量统计
     *
     * @return
     */
    @GetMapping("/getMemberReport")
    public Result getMemberReport() {
        Date today = DateUtils.getToday ( );
        Map<String, Object> memberReport = reportService.getMemberReport ( );
        return Result.success (memberReport, "获取会员统计数据成功");
    }

    //运营数据统计
    @RequestMapping("/getBusinessReportData")
    public Result getBusinessReportData() {
        try {
            Map<String, Object> data = reportService.getBusinessReportData ( );
            return Result.success (data, MessageConstant.GET_BUSINESS_REPORT_SUCCESS);
        } catch (Exception e) {
            return Result.error (MessageConstant.GET_BUSINESS_REPORT_FAIL);
        }
    }

    //导出运营数据
    @RequestMapping("/exportBusinessReport")
    public Result exportBusinessReport(HttpSession session, HttpServletResponse response) {
        try {
            Map<String, Object> result = reportService.getBusinessReportData ( );
            //取出返回结果数据，准备将报表数据写入到Excel文件中
            String reportDate = (String) result.get ("reportDate");
            Integer todayNewMember = (Integer) result.get ("todayNewMember");
            Integer totalMember = (Integer) result.get ("totalMember");
            Integer thisWeekNewMember = (Integer) result.get ("thisWeekNewMember");
            Integer thisMonthNewMember = (Integer) result.get ("thisMonthNewMember");
            Integer todayOrderNumber = (Integer) result.get ("todayOrderNumber");
            Integer thisWeekOrderNumber = (Integer) result.get ("thisWeekOrderNumber");
            Integer thisMonthOrderNumber = (Integer) result.get ("thisMonthOrderNumber");
            Integer todayVisitsNumber = (Integer) result.get ("todayVisitsNumber");
            Integer thisWeekVisitsNumber = (Integer) result.get ("thisWeekVisitsNumber");
            Integer thisMonthVisitsNumber = (Integer) result.get ("thisMonthVisitsNumber");
            List<Map> hotSetmeal = (List<Map>) result.get ("hotSetmeal");

//            String filePath = session.getServletContext ( ).getRealPath ("template") + File.separator + "report_template.xlsx";
            //基于提供的Excel模板文件在内存中创建一个Excel表格对象
//            XSSFWorkbook excel = new XSSFWorkbook (new FileInputStream (new File (filePath)));


            File file = ResourceUtils.getFile ("classpath:static" + File.separator + "template" + File.separator + "report_template.xlsx");
            XSSFWorkbook excel = new XSSFWorkbook (new FileInputStream (file));
            //读取第一个工作表
            XSSFSheet sheet = excel.getSheetAt (0);

            XSSFRow row = sheet.getRow (2);
            row.getCell (5).setCellValue (reportDate);//日期

            row = sheet.getRow (4);
            row.getCell (5).setCellValue (todayNewMember);//新增会员数（本日）
            row.getCell (7).setCellValue (totalMember);//总会员数

            row = sheet.getRow (5);
            row.getCell (5).setCellValue (thisWeekNewMember);//本周新增会员数
            row.getCell (7).setCellValue (thisMonthNewMember);//本月新增会员数

            row = sheet.getRow (7);
            row.getCell (5).setCellValue (todayOrderNumber);//今日预约数
            row.getCell (7).setCellValue (todayVisitsNumber);//今日到诊数

            row = sheet.getRow (8);
            row.getCell (5).setCellValue (thisWeekOrderNumber);//本周预约数
            row.getCell (7).setCellValue (thisWeekVisitsNumber);//本周到诊数

            row = sheet.getRow (9);
            row.getCell (5).setCellValue (thisMonthOrderNumber);//本月预约数
            row.getCell (7).setCellValue (thisMonthVisitsNumber);//本月到诊数

            int rowNum = 12;
            for (Map map : hotSetmeal) {//热门套餐
                String name = (String) map.get ("name");
                Long setmeal_count = (Long) map.get ("setmeal_count");
                BigDecimal proportion = (BigDecimal) map.get ("proportion");
                row = sheet.getRow (rowNum++);
                row.getCell (4).setCellValue (name);//套餐名称
                row.getCell (5).setCellValue (setmeal_count);//预约数量
                row.getCell (6).setCellValue (proportion.doubleValue ( ));//占比
            }

            //使用输出流进行表格下载,基于浏览器作为客户端下载
            OutputStream out = response.getOutputStream ( );
            response.setContentType ("application/vnd.ms-excel");//代表的是Excel文件类型 content-type  application/vnd.ms-excel
            // response.setHeader("Content-Type","application/vnd.ms-excel");
            response.setHeader ("content-Disposition", "attachment;filename=report.xlsx");//指定以附件形式进行下载 强制下载！！
            excel.write (out);

            out.flush ( );
            out.close ( );
            excel.close ( );
            return null;
        } catch (Exception e) {
            return Result.error (MessageConstant.GET_BUSINESS_REPORT_FAIL);
        }
    }


    @Autowired
    private OrderService orderService;

    @RequestMapping("/getOrderAndVisit")
    public Result getOrderAndVisit(String begin, String end) {

        Map<String, Object> map = null;
        try {
            map = orderService.getOrderAndVisit (begin, end);
        } catch (Exception e) {
            e.printStackTrace ( );
            return Result.error (MessageConstant.GET_ORDER_COUNT_FAIL);
        }
        return Result.success (map, MessageConstant.GET_ORDER_COUNT_SUCCESS);
    }

    //套餐占比分析
    @RequestMapping("/getSetmealReport")
    public Result getSetmealReport() {
        Map setmealReport = reportService.getSetmealReport ( );
        return Result.success (setmealReport, MessageConstant.GET_SETMEAL_COUNT_REPORT_SUCCESS);
    }

}
